import type { MarkType, Schema } from "@rivet/model";

//获取标记类型
export function getMarkType(
  nameOrType: string | MarkType,
  schema: Schema,
): MarkType {
  if (typeof nameOrType === "string") {
    if (!schema.marks[nameOrType]) {
      throw Error(`没有标记类型 '${nameOrType}'. 可能是忘记添加扩展了?`);
    }

    return schema.marks[nameOrType];
  }

  return nameOrType;
}
